{%- for message in messages %}
{%- if message['role'] == 'user' %}
    {{- '<|user|>' + message['content'] + eos_token }}
{%- elif message['role'] == 'system' %}
    {{- '<|system|>' + message['content'] + eos_token }}
{%- elif message['role'] == 'assistant' %}
    {{- '<|assistant|>'  + message['content'] + eos_token }}
{%- endif %}
{%- if loop.last and add_generation_prompt %}
    {{- '<|assistant|>' }}
{%- endif %}
{%- endfor %}